Spark Machine Learning Project (Employee Attrition Prediction) for beginners using Databricks Notebook (Unofficial) (Community edition Server) In this Data science Machine Learning project, we will create Employee Attrition Prediction Project using Decision Tree Classification algorithm one of the predictive models.

Get your team access to 3,500 top Udemy courses anytime, anywhere. In this Data science Machine Learning project, we will create Employee Attrition Prediction Project using Decision Tree Classification algorithm one of the predictive models. Databricks lets you start writing Spark ML code instantly so you can focus on your data problems.

We aim to predict whether an employee of a company will leave or not, using the k-Nearest Neighbors algorithm. We use evaluation of employee performance, average monthly hours at work and number of years spent in the company, among others, as our features. Other approaches to this problem include the use of ANNs, decision trees and logistic regression. The dataset was split, using 70% for training the algorithm and 30% for testing it, achieving an accuracy of 94.32%.